import torch
from config import config
from datam import *
from model import Model,Backbone
from torch.nn import functional as F

if __name__ == '__main__':


    data_mnist = DataM(config.data_dir, 1, config.AVAIL_GPUS,224)
    data_mnist.setup()
    images, labels = next(iter(data_mnist.train_dataloader()))
    print(images.shape,labels)
    T.ToPILImage()(images[0]).show()



